import os
import torch
from EAattak.Attck import Attck
from configure import configure
from data_utils import Dataer
from network import Network
import numpy as np
import time


config = vars(configure())

device = 'cuda' if torch.cuda.is_available() else 'cpu'
config['device'] = device

dataer = Dataer(dataset_name=config['dataset'], block_size=config['block_size'])

shuffle_loader = dataer.get_shuffle_loader(seed=0, batch_size=1, isTrain=False)

loader = dataer.get_cross_class_loader(0, 80, isTrain=True, batch_size=16)

sample, _ = next(iter(loader))